# -*- coding: utf-8 -*-

"""
@File    : 01_栅格文件展示.py
@Author  : fungis@163.com
@Time    : 2022/09/17 23:32
@notice  :
@notice  :
"""
from matplotlib import pyplot as plt
from osgeo import gdal  # 或者直接用import gdal

raster_path = r'./data-use/tif/AP_05726_FBS_F0680_RT1.dem.tif'

# 读取影像
dataset = gdal.Open(raster_path)
# 打开波段1（注意:用索引1，而不是0，来获取第一个波段）
band = dataset.GetRasterBand(1)
elevation = band.ReadAsArray()
print(elevation.shape)  # 行列数
print(elevation)  # 行列矩阵

plt.imshow(elevation, cmap='gist_earth')  # cmap='gist_earth'以gist_earth色带形式展示
plt.show()
